package com.ibm.iisp.common.util;

import java.io.UnsupportedEncodingException;

import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;

import org.apache.commons.codec.binary.Base64;

/**
 * 
 * AES算法封装
 */
public class AESEncryptUtil {

	/**
	 * 加密算法
	 */
	private static final String ENCRY_ALGORITHM = "AES";

	/**
	 * 加密算法/加密模式/填充类型 本例采用AES加密，ECB加密模式，PKCS5Padding填充
	 */
	private static final String CIPHER_MODE = "AES/ECB/PKCS5Padding";

	/*
	 * 设置iv偏移量 本例采用ECB加密模式，不需要设置iv偏移量
	 */
	// private static final String IV_ = null;

	/**
	 * 设置加密字符集 本例采用 UTF-8 字符集
	 */
	private static final String CHARACTER = "UTF-8";

	/**
	 * 设置加密密码处理长度。 不足此长度补0；
	 */
	private static final int PWD_SIZE = 16;

	/**
	 * 密码处理方法 如果加解密出问题， 请先查看本方法，排除密码长度不足补"0",导致密码不一致
	 * 
	 * @param password 待处理的密码
	 * @return
	 * @throws UnsupportedEncodingException
	 */
	private static byte[] pwdHandler(String password) throws UnsupportedEncodingException {
		byte[] data = null;
		if (password == null) {
			password = "";
		}
		StringBuffer sb = new StringBuffer(PWD_SIZE);
		sb.append(password);
		while (sb.length() < PWD_SIZE) {
			sb.append("0");
		}
		if (sb.length() > PWD_SIZE) {
			sb.setLength(PWD_SIZE);
		}

		data = sb.toString().getBytes("UTF-8");

		return data;
	}

	// ======================>原始加密<======================

	/**
	 * 原始加密
	 * 
	 * @param clearTextBytes 明文字节数组，待加密的字节数组
	 * @param pwdBytes       加密密码字节数组
	 * @return 返回加密后的密文字节数组，加密错误返回null
	 * @throws Exception any exception
	 */
	public static byte[] encrypt(byte[] clearTextBytes, byte[] pwdBytes) throws Exception {
		// 1 获取加密密钥
		SecretKeySpec keySpec = new SecretKeySpec(pwdBytes, ENCRY_ALGORITHM);

		// 2 获取Cipher实例
		Cipher cipher = Cipher.getInstance(CIPHER_MODE);

		// 查看数据块位数 默认为16（byte） * 8 =128 bit
//            System.out.println("数据块位数(byte)：" + cipher.getBlockSize());

		// 3 初始化Cipher实例。设置执行模式以及加密密钥
		cipher.init(Cipher.ENCRYPT_MODE, keySpec);

		// 4 执行
		byte[] cipherTextBytes = cipher.doFinal(clearTextBytes);

		// 5 返回密文字符集
		return cipherTextBytes;

	}

	/**
	 * 原始解密
	 * 
	 * @param cipherTextBytes 密文字节数组，待解密的字节数组
	 * @param pwdBytes        解密密码字节数组
	 * @return 返回解密后的明文字节数组，解密错误返回null
	 * @throws Exception any exception
	 */
	public static byte[] decrypt(byte[] cipherTextBytes, byte[] pwdBytes) throws Exception {

		// 1 获取解密密钥
		SecretKeySpec keySpec = new SecretKeySpec(pwdBytes, ENCRY_ALGORITHM);

		// 2 获取Cipher实例
		Cipher cipher = Cipher.getInstance(CIPHER_MODE);

		// 查看数据块位数 默认为16（byte） * 8 =128 bit
//            System.out.println("数据块位数(byte)：" + cipher.getBlockSize());

		// 3 初始化Cipher实例。设置执行模式以及加密密钥
		cipher.init(Cipher.DECRYPT_MODE, keySpec);

		// 4 执行
		byte[] clearTextBytes = cipher.doFinal(cipherTextBytes);

		// 5 返回明文字符集
		return clearTextBytes;

	}

	// ======================>BASE64<======================

	/**
	 * BASE64加密
	 * 
	 * @param clearText 明文，待加密的内容
	 * @param password  密码，加密的密码
	 * @return 返回密文，加密后得到的内容。加密错误返回null
	 * @throws Exception any exception
	 */
	public static String encryptBase64(String clearText, String password) throws Exception {
		// 1 获取加密密文字节数组
		byte[] cipherTextBytes = encrypt(clearText.getBytes(CHARACTER), pwdHandler(password));

		// 2 对密文字节数组进行BASE64 encoder 得到 BASE6输出的密文
		String cipherText = Base64.encodeBase64String(cipherTextBytes);

		// 3 返回BASE64输出的密文
		return cipherText;
	}

	/**
	 * BASE64解密
	 * 
	 * @param cipherText 密文，带解密的内容
	 * @param password   密码，解密的密码
	 * @return 返回明文，解密后得到的内容。解密错误返回null
	 * @throws Exception any exception
	 */
	public static String decryptBase64(String cipherText, String password) throws Exception {
		// 1 对 BASE64输出的密文进行BASE64 decodebuffer 得到密文字节数组
		byte[] cipherTextBytes = Base64.decodeBase64(cipherText);

		// 2 对密文字节数组进行解密 得到明文字节数组
		byte[] clearTextBytes = decrypt(cipherTextBytes, pwdHandler(password));

		// 3 根据 CHARACTER 转码，返回明文字符串
		return new String(clearTextBytes, CHARACTER);
	}

	// ======================>HEX<======================

	/**
	 * HEX加密
	 * 
	 * @param clearText 明文，待加密的内容
	 * @param password  密码，加密的密码
	 * @return 返回密文，加密后得到的内容。加密错误返回null
	 * @throws Exception any exception
	 */
	public static String encryptHex(String clearText, String password) throws Exception {
		// 1 获取加密密文字节数组
		byte[] cipherTextBytes = encrypt(clearText.getBytes(CHARACTER), pwdHandler(password));

		// 2 对密文字节数组进行 转换为 HEX输出密文
		String cipherText = byte2hex(cipherTextBytes);

		// 3 返回 HEX输出密文
		return cipherText;
	}

	/**
	 * HEX解密
	 * 
	 * @param cipherText 密文，带解密的内容
	 * @param password   密码，解密的密码
	 * @return 返回明文，解密后得到的内容。解密错误返回null
	 * @throws Exception any exception
	 */
	public static String decryptHex(String cipherText, String password) throws Exception {
		// 1 将HEX输出密文 转为密文字节数组
		byte[] cipherTextBytes = hex2byte(cipherText);

		// 2 将密文字节数组进行解密 得到明文字节数组
		byte[] clearTextBytes = decrypt(cipherTextBytes, pwdHandler(password));

		// 3 根据 CHARACTER 转码，返回明文字符串
		return new String(clearTextBytes, CHARACTER);
	}

	/* 字节数组转成16进制字符串 */
	static String byte2hex(byte[] bytes) { // 一个字节的数，
		StringBuffer sb = new StringBuffer(bytes.length * 2);
		String tmp = "";
		for (int n = 0; n < bytes.length; n++) {
			// 整数转成十六进制表示
			tmp = (java.lang.Integer.toHexString(bytes[n] & 0XFF));
			if (tmp.length() == 1) {
				sb.append("0");
			}
			sb.append(tmp);
		}
		return sb.toString().toUpperCase(); // 转成大写
	}

	/* 将hex字符串转换成字节数组 */
	private static byte[] hex2byte(String str) {
		if (str == null || str.length() < 2) {
			return new byte[0];
		}
		str = str.toLowerCase();
		int l = str.length() / 2;
		byte[] result = new byte[l];
		for (int i = 0; i < l; ++i) {
			String tmp = str.substring(2 * i, 2 * i + 2);
			result[i] = (byte) (Integer.parseInt(tmp, 16) & 0xFF);
		}
		return result;
	}

//	public static void main(String[] args) throws Exception {
//		String test = encryptHex("admin", "123456");
//		System.out.println(test);
//
//		System.out.println(decryptHex(test, "123456"));
//	}
}